visualizing python code flow